package com.leetcode.array;

import com.leetcode.annotation.Remark;

import java.util.Arrays;

/**
 * 870. 优势洗牌
 * 给定两个大小相等的数组 nums1 和 nums2，nums1 相对于 nums2 的优势可以用满足 nums1[i] > nums2[i] 的索引 i 的数目来描述。
 * <p>
 * 返回 nums1 的任意排列，使其相对于 nums2 的优势最大化。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums1 = [2,7,11,15], nums2 = [1,10,4,11]
 * 输出：[2,11,7,15]
 * 示例 2：
 * <p>
 * 输入：nums1 = [12,24,8,32], nums2 = [13,25,32,11]
 * 输出：[24,32,8,12]
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= nums1.length <= 105
 * nums2.length == nums1.length
 * 0 <= nums1[i], nums2[i] <= 109
 *
 * https://leetcode.cn/problems/advantage-shuffle/
 */
@Remark(" - 870. 优势洗牌")
public class M_advantageCount {
    public static void main(String[] args) {
        int[] nums1 = {2, 7, 11, 15}, nums2 = {1, 100, 40, 110};
        System.out.println(advantageCount(nums1, nums2));

    }

    public static int[] advantageCount(int[] nums1, int[] nums2) {
        int len = nums1.length, sidx = 0;
        boolean[] check = new boolean[len];
        int[] ans = new int[len];
        Arrays.sort(nums1);
        for (int i = 0; i < len; i += 1) {
            int le = 0, ri = len, mid, tar = nums2[i];
            while (le != ri) if (nums1[(mid = (le + ri) >> 1)] < tar) le = mid + 1;
            else ri = mid;
            while (le < len && (check[le] || nums1[le] == tar)) le += 1;
            if (le == len) {
                while (check[sidx]) sidx += 1;
                ans[i] = nums1[sidx];
                check[sidx++] = true;
            } else {
                check[le] = true;
                ans[i] = nums1[le];
            }
        }
        return ans;
    }
}
